﻿// Spun AJAX Framework
// Copyright (C) 2008 Blake Hegerle
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;

namespace Spun.Applications
{
    public class ThreadToApplicationContextAssociation : IDisposable
    {
        public ThreadToApplicationContextAssociation(ApplicationContext ApplicationContext)
        {
            this.ApplicationContext = ApplicationContext;

            Id = Thread.CurrentThread.ManagedThreadId;

            lock (Sync)
                ContextMap[Id] = ApplicationContext;
        }

        public static ApplicationContext ApplicationContextAssociatedWithCurrentThread
        {
            get
            {
                var Id = Thread.CurrentThread.ManagedThreadId;

                lock (Sync)
                {
                    ApplicationContext AppCtx;
                    ContextMap.TryGetValue(Id, out AppCtx);
                    return AppCtx;
                }
            }
        }

        #region IDisposable Members

        public void Dispose()
        {
            lock (Sync)
                if (ContextMap.ContainsKey(Id))
                    ContextMap.Remove(Id);
        }

        #endregion

        private ApplicationContext ApplicationContext;
        private int Id;

        private static object Sync;
        private static IDictionary<int, ApplicationContext> ContextMap;

        static ThreadToApplicationContextAssociation()
        {
            Sync = new object();
            ContextMap = new Dictionary<int, ApplicationContext>();
        }
    }
}
